package BinarySortTree;

/*
二叉搜索树中的搜索
给定二叉搜索树（BST）的根节点root和一个整数值val。
你需要在 BST 中找到节点值等于val的节点。 返回以该节点为根的子树。 如果节点不存在，则返回null。

示例 1:
输入：root = [4,2,7,1,3], val = 2
输出：[2,1,3]
示例 2:

输入：root = [4,2,7,1,3], val = 5
输出：[]

作者：LeetCode
链接：https://leetcode.cn/leetbook/read/introduction-to-data-structure-binary-search-tree/xpsqtv/
 */

public class _21二叉搜索树中的搜索 {
    public static void main(String[] args) {

    }

    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode() {
        }

        TreeNode(int val) {
            this.val = val;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }

    public TreeNode searchBST(TreeNode root, int val) {
        if (root == null) {
            return null;
        }
        return dfs(root,val);
    }

    public TreeNode dfs(TreeNode root, int val) {
        if (root == null) {
            return null;
        }
        if (root.val == val) {
            return root;
        } else if (root.val > val) {
            return dfs(root.left, val);
        }else {
            return dfs(root.right, val);
        }
    }

    //官解：递归
    class Solution {
        public TreeNode searchBST(TreeNode root, int val) {
            if (root == null) {
                return null;
            }
            if (val == root.val) {
                return root;
            }
            return searchBST(val < root.val ? root.left : root.right, val);
        }
    }

    //官解：迭代
    class Solution2 {
        public TreeNode searchBST(TreeNode root, int val) {
            while (root != null) {
                if (val == root.val) {
                    return root;
                }
                root = val < root.val ? root.left : root.right;
            }
            return null;
        }
    }

}
